package com.mj.listen3.链表;

/**
 * 删除链表的中间节点和a/b处的节点
 */
public class _0_删除链表的中间节点和a除以b处的节点 {

    /**
     * 删除 a/b 位置的节点。
     */
    public ListNode removeByRatio(ListNode head, int a, int b) {
        if (a < 1 || a > b) {
            return head;
        }

        // 找出链表总节点个数
        int n = 0;
        ListNode cur = head;
        while (cur != null) {
            n++;
            cur = cur.next;
        }

        // 求出删除第几个节点，复用n变量
        n = (int) Math.ceil(((double) (a * n)) / (double) b);

        if (n == 1) {
            // 表示删除的是头节点
            head = head.next;
        }

        // 删除的是内部节点，找出删除节点的前一个节点
        if (n > 1) {
            cur = head;
            while (--n != 1) {
                // 依次找出要删除节点的前一个节点
                cur = cur.next;
            }
            cur.next = cur.next.next;
        }

        return head;
    }

    /**
     * 删除中间节点
     */
    public ListNode removeMidNode(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }

        if (head.next.next == null) {
            return head.next;
        }

        // 超出3个节点
        // 找出要删除节点的前一个节点
        ListNode pre = head;
        ListNode cur = head.next.next;
        while (cur.next != null && cur.next.next != null) {
            pre = pre.next;
            cur = cur.next.next;
        }
        pre.next = pre.next.next;
        return head;
    }
}
